// middleware/database.go
package middleware

import (
	"github.com/gin-gonic/gin"
	"gorm.io/gorm"
)

// DatabaseMiddleware 返回一个将数据库连接注入到上下文的中间件
func DatabaseMiddleware(db *gorm.DB) gin.HandlerFunc {
	return func(c *gin.Context) {
		// 将数据库连接放入请求上下文
		c.Set("db", db)

		// 继续处理请求
		c.Next()
	}
}
